package com.gpframework.common.utils;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.ArrayList;
import java.util.List;

/**
 * 精确的浮点数运算
 *
 * @Author gpframework
 */
public class Arith
{

    /** 默认除法运算精度 */
    private static final int DEF_DIV_SCALE = 10;

    private static final BigDecimal ZERO = BigDecimal.ZERO;


    /** 这个类不能实例化 */
    private Arith()
    {
    }

    /**
     * 提供精确的加法运算。
     * @param v1 被加数
     * @param v2 加数
     * @return 两个参数的和
     */
    public static double add(double v1, double v2)
    {
        BigDecimal b1 = BigDecimal.valueOf(v1);
        BigDecimal b2 = BigDecimal.valueOf(v2);
        return b1.add(b2).doubleValue();
    }

    /**
     * 提供精确的加法运算。
     * @param v1 被加数
     * @param v2 加数
     * @return 两个参数的和
     */
    public static BigDecimal add(BigDecimal... params)
    {
        BigDecimal result = BigDecimal.ZERO;
        for (BigDecimal param : params) {
            result = result.add(param).setScale(2, RoundingMode.HALF_UP);
        }
        return result;
    }


    /**
     * 提供精确的减法运算。
     * @param v1 被减数
     * @param v2 减数
     * @return 两个参数的差
     */
    public static double sub(double v1, double v2)
    {
        BigDecimal b1 = BigDecimal.valueOf(v1);
        BigDecimal b2 = BigDecimal.valueOf(v2);
        return b1.subtract(b2).doubleValue();
    }

    /**
     * 提供精确的减法运算。
     * @param v1 被减数
     * @param v2 减数
     * @return 两个参数的差 第一个参数为被减数，其余数字为减数
     */
    public static BigDecimal sub(BigDecimal... params)
    {
        BigDecimal result = params[0];
        for (int i = 1; i < params.length; i++) {
            result = result.subtract(params[i]).setScale(2, RoundingMode.HALF_UP);
        }
        return result;
    }

    /**
     * 提供精确的乘法运算。
     * @param v1 被乘数
     * @param v2 乘数
     * @return 两个参数的积
     */
    public static double mul(double v1, double v2)
    {
        BigDecimal b1 = BigDecimal.valueOf(v1);
        BigDecimal b2 = BigDecimal.valueOf(v2);
        return b1.multiply(b2).doubleValue();
    }

    /**
     * 提供精确的乘法运算。
     * @param v1 被乘数
     * @param v2 乘数
     * @return 两个参数的积
     */
    public static BigDecimal mul(BigDecimal b1, BigDecimal b2)
    {
        return b1.multiply(b2);
    }

    /**
     * 提供（相对）精确的除法运算，当发生除不尽的情况时，精确到
     * 小数点以后10位，以后的数字四舍五入。
     * @param v1 被除数
     * @param v2 除数
     * @return 两个参数的商
     */
    public static BigDecimal div(BigDecimal v1, BigDecimal v2)
    {
        return div(v1, v2, 2);
    }

    /**
     * 提供（相对）精确的除法运算。当发生除不尽的情况时，由scale参数指
     * 定精度，以后的数字四舍五入。
     * @param v1 被除数
     * @param v2 除数
     * @param scale 表示表示需要精确到小数点以后几位。
     * @return 两个参数的商
     */
    public static double div(double v1, double v2, int scale)
    {
        if (scale < 0)
        {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b1 = BigDecimal.valueOf(v1);
        BigDecimal b2 = BigDecimal.valueOf(v2);
        if (b1.compareTo(BigDecimal.ZERO) == 0)
        {
            return BigDecimal.ZERO.doubleValue();
        }
        return b1.divide(b2, scale, RoundingMode.HALF_UP).doubleValue();
    }

    /**
     * 提供（相对）精确的除法运算。当发生除不尽的情况时，由scale参数指
     * 定精度，以后的数字四舍五入。
     * @param v1 被除数
     * @param v2 除数
     * @param scale 表示表示需要精确到小数点以后几位。
     * @return 两个参数的商
     */
    public static BigDecimal div(BigDecimal b1, BigDecimal b2, int scale)
    {
        if (scale < 0)
        {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        if (b1.compareTo(BigDecimal.ZERO) == 0)
        {
            return BigDecimal.ZERO;
        }
        return b1.divide(b2, scale, RoundingMode.HALF_UP);
    }

    /**
     * 提供精确的小数位四舍五入处理。
     * @param v 需要四舍五入的数字
     * @param scale 小数点后保留几位
     * @return 四舍五入后的结果
     */
    public static double round(double v, int scale)
    {
        if (scale < 0)
        {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b = BigDecimal.valueOf(v);
        BigDecimal one = new BigDecimal("1");
        return b.divide(one, scale, RoundingMode.HALF_UP).doubleValue();
    }

    /**
     * 提供精确的小数位四舍五入处理。
     * @param v 需要四舍五入的数字
     * @param scale 小数点后保留几位
     * @return 四舍五入后的结果
     */
    public static BigDecimal round(BigDecimal b, int scale)
    {
        if (scale < 0)
        {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal one = new BigDecimal("1");
        return b.divide(one, scale, RoundingMode.HALF_UP);
    }

    public static List<BigDecimal> averageDiscount(BigDecimal discountAmount, int itemCount) {
        List<BigDecimal> list = new ArrayList<>();
        if (discountAmount == null || discountAmount.intValue() == 0 || itemCount <= 0){
            return null;
        }
        BigDecimal averageDiscount = discountAmount.divide(BigDecimal.valueOf(itemCount), 2, BigDecimal.ROUND_HALF_UP);
        BigDecimal remainingDiscount = discountAmount.subtract(averageDiscount.multiply(BigDecimal.valueOf(itemCount)));
        for (int i = 0; i < itemCount; i++) {
            BigDecimal allocatedDiscount = averageDiscount;
            if (remainingDiscount.compareTo(BigDecimal.ZERO) != 0) {
                allocatedDiscount = Arith.add(allocatedDiscount,remainingDiscount);
                remainingDiscount = BigDecimal.ZERO;
            }
            list.add(allocatedDiscount);
        }
        return list;
    }

    /**
     * 百分比转小数
     * @param price
     * @return
     */
    public static BigDecimal ratio(BigDecimal price) {
        return div(price,BigDecimal.valueOf(100));

    }
    /**
     * 获取金额
     * @param price
     * @return
     */
    public static BigDecimal of(BigDecimal price) {
        return price == null ? ZERO : price;
    }
    /**
     * 获取金额
     * @param integer
     * @return
     */
    public static BigDecimal of(Integer integer) {
        return integer == null ? ZERO : BigDecimal.valueOf(integer);
    }
    /**
     * 获取金额
     * @param l
     * @return
     */
    public static BigDecimal of(Long l) {
        return l == null ? ZERO : BigDecimal.valueOf(l);
    }
    public static BigDecimal of(Double d) {
        return d == null ? ZERO : BigDecimal.valueOf(d);
    }
    /**
     * 获取金额
     * @param str
     * @return
     */
    public static BigDecimal of(String str) {
        return StringUtils.isBlank(str) ? ZERO : new BigDecimal(str);
    }
}
